class Solution {
    int[][][] dp;
    int[] boxes;
    public int removeBoxes(int[] _boxes) {
        dp = new int[101][1001][101];
        boxes = _boxes;
        return dfs(0, boxes.length - 1, 0);
    }
    public int dfs(int l, int r, int k) {
        if (l > r) {
            return 0;
        }
        while (r - 1 >= l && boxes[r] == boxes[r - 1]) {
            r--;
            k++;
        }
        if (dp[l][r][k] != 0) {
            return dp[l][r][k];
        }
        // 选择1 直接删
        dp[l][r][k] = dfs(l, r - 1, 0) + (k + 1) * (k + 1);
        // 选择2, 存起来，找前面的相同数字，一切删；
        for (int i = l; i < r; i++) {
            if (boxes[i] == boxes[r]) {
                dp[l][r][k] = Math.max(dp[l][r][k], dfs(l, i, k + 1) + dfs(i + 1, r - 1, 0));
            }
        }
        return dp[l][r][k];
    }
}

